// Copyright (c) 2025, 大连理工大学 (Dalian University of Technology)
//
// Licensed under the Mulan PSL v2.
// You can use this file according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of the License at
//
//     http://license.coscl.org.cn/MulanPSL2
//
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. SEE THE MULAN PSL v2
// FOR MORE DETAILS.

#include <gtest/gtest.h>

#include "Utility/FileSystem/SGFileSystem.h"

#include "DBManager/DBServiceFactory.h"
#include "DBManager/Database/DatabaseFactory.h"

#include "Import/ImportAPI/ImportAPI.h"
using namespace SG::DBManager;
using namespace SG::Import;
using namespace SG::DataStructure;
/// @brief 数据库服务工厂
std::shared_ptr<DBServiceFactory> DB_SERVICEFACTORY = nullptr;

class H5DBEnvironment : public testing::Environment
{
  public:
    DatabaseFactory* m_dbFactory = nullptr;

    void SetUp () override
    {
        // 数据库工厂
        m_dbFactory = new DatabaseFactory ();
        // 数据库
        IDatabase* db = m_dbFactory->GetDatabase (DatabaseFactory::HDF5, ".", "ImportUnitTest.db");
        // 根据数据库创建Service管理器
        DB_SERVICEFACTORY = std::make_shared<DBServiceFactory> (db);
    }
    void TearDown () override
    {
        // 释放数据库资源
        DB_SERVICEFACTORY.reset ();
        delete m_dbFactory;
    }
};

int main (int argc, char* argv[])
{
    testing::AddGlobalTestEnvironment (new H5DBEnvironment ());
    testing::InitGoogleTest (&argc, argv);
    return RUN_ALL_TESTS ();
}
